import torch

# 定义模型标识符和保存路径
save_path = "unet_carvana.pt"

net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana', pretrained=True, scale=0.5)

# 保存模型到本地
torch.save(net, save_path)
